#include "qwt3d_axis.h"

using namespace Qwt3D;

Axis::Axis()
{
  init();
};

Axis::~Axis()
{
}

Axis::Axis(Triple beg, Triple end)
{
  init();
  setPosition(beg,end);
}

void Axis::init()
{
  detachAll();

  scale_ = ValuePtr<Scale>(new LinearScale);

  beg_ = Triple(0.0, 0.0, 0.0);  
  end_ = beg_;
  
  majorintervals_ = 0;
  minorintervals_ = 0;
  setMajors(1);   
  setMinors(1);   
  setLimits(0,0);

  setTicOrientation(0.0, 0.0, 0.0);
  setTicLength(0.0, 0.0);
  setColor(0.0, 0.0, 0.0);
  setLineWidth(1.0);
  symtics_ = false;
  drawNumbers_ = false;
  drawLabel_ = false;

  drawTics_ = false;
  autoscale_ = true;
  markerLabel_.clear();
  numberfont_ = QFont("Courier",12);
  setLabelFont(QFont("Courier",14));

  numbercolor_ = RGBA(0,0,0,0);

  setNumberAnchor(Center);

  numbergap_ = 0;
  labelgap_ = 0;

  isZ_ = false;
}

void Axis::setPosition(const Triple& beg, const Triple& end)
{
  beg_ = beg;
  end_ = end;
}

void Axis::setMajors(int val)
{
  if (val == majorintervals_)
    return;
  
  majorintervals_ = (val<=0) ? 1 : val; // always >= 1
}

/*!
\see LogScale::setMinors().
*/
void Axis::setMinors(int val)
{
  if (val == minorintervals_)
    return;

  minorintervals_ = (val<=0) ? 1 : val; // always >= 1
}

void Axis::setTicLength(double majorl, double minorl)
{
  lmaj_ = majorl;
  lmin_ = minorl;
}

void Axis::setTicOrientation(double tx, double ty, double tz)
{
  setTicOrientation(Triple(tx,ty,tz));
}

void Axis::setTicOrientation(const Triple& val)
{
  orientation_ = val;
  orientation_.normalize();
}

/**
\param val thickness for axis base line
\param majfac relative thickness for axis major tics (majfac*val)
\param minfac relative thickness for axis minor tics (minfac*val)
*/
void Axis::setLineWidth(double val, double majfac, double minfac)
{
  lineWidth_ = val;
  majLineWidth_ = majfac * lineWidth_;
  minLineWidth_ = minfac * lineWidth_;
}

void Axis::draw()
{
  Drawable::draw();

  saveGLState();

//  GLStateBewarer sb(GL_LINE_SMOOTH, true);
//  glBlendFunc(GL_ONE, GL_ZERO);
  glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
  glColor4d(color.r,color.g,color.b,color.a);     

  drawBase();
  drawTics();
  drawLabel();    

  restoreGLState();
}

/**
Always use AFTER drawNumbers() ! (Needs length of number string)
*/
void Axis::drawLabel()
{
  if (!drawLabel_)
      return;

	double width = 0.0;
	for (unsigned i = 0; i != markerLabel_.size(); i++){
		double aux = markerLabel_[i].width();
		if (aux > width)
			width = aux;
	}

	Triple center = begin() + (end_ - beg_)/2;
	Triple ticEnd = ViewPort2World(World2ViewPort(center + ticOrientation() * lmaj_));

	double rap = (width + labelgap_ + label_.textHeight())/(World2ViewPort(ticEnd) - World2ViewPort(center)).length();

	Triple pos = ViewPort2World(World2ViewPort(center + ticOrientation() * lmaj_*(1 + rap)));
	setLabelPosition(pos, Center);

	Triple end = World2ViewPort(end_);
	Triple beg = World2ViewPort(beg_);
	double angle = 360 - fabs(QLineF(beg.x, beg.y, end.x, end.y).angle());

  if (!isZ()) 
  {
    if (angle > 90 && angle < 180)
	    angle += 180;
    if (angle > 180 && angle < 270)
	    angle -= 180;
	}

	label_.draw(angle);
}

void Axis::drawBase()
{
  setLineWidth( lineWidth_ );
  glBegin( GL_LINES );
    glVertex3d( beg_.x, beg_.y, beg_.z); 
    glVertex3d( end_.x, end_.y, end_.z);
  glEnd();
}   

bool Axis::prepTicCalculation(Triple& startpoint)
{
  if (isPracticallyZero(start_, stop_))
    return false;

  autostart_ = start_;
  autostop_ = stop_;

  if (autoScale()) 
  {  
    setMajors(scale_->autoscale(autostart_, autostop_, start_, stop_, majors()));
    if (isPracticallyZero(autostart_, autostop_))
      return false;
  }
  
  scale_->setLimits(start_,stop_);
  scale_->setMajors(majors());
  scale_->setMinors(minors());
  scale_->setMajorLimits(autostart_,autostop_);
  scale_->calculate();

  Triple normal = (end_ - beg_);
  //normal.normalize();
  Triple beg = beg_ + ((autostart_ - start_) / (stop_ - start_)) * normal;
  Triple end = end_ - ((stop_ - autostop_) / (stop_ - start_))* normal;

  startpoint = end_ - beg_;

  majorpos_.clear();
  minorpos_.clear();

  return true;
}

void Axis::recalculateTics()
{
  Triple runningpoint;
  if (false==prepTicCalculation(runningpoint))
    return;

  unsigned int i;
  
  for (i = 0; i != scale_->majors_p.size(); ++i) 
  {
    double t = (scale_->majors_p[i] - start_) / (stop_-start_);
    majorpos_.push_back(beg_ + t * runningpoint);
  }
  for (i = 0; i != scale_->minors_p.size(); ++i) 
  {
    double t = (scale_->minors_p[i] - start_) / (stop_-start_);
    minorpos_.push_back(beg_ + t * runningpoint);
  }
}

void Axis::drawTics()
{
  Triple runningpoint;
  if (!drawTics_ || false==prepTicCalculation(runningpoint))
    return;
  
  unsigned int i;
  Triple nadir;
    
  markerLabel_.resize(scale_->majors_p.size());
  setLineWidth(majLineWidth_);
  for (i = 0; i != scale_->majors_p.size(); ++i) 
  {
    double t = (scale_->majors_p[i] - start_) / (stop_-start_);
    nadir = beg_ + t * runningpoint;
    majorpos_.push_back(drawTic(nadir, lmaj_));
    drawTicLabel(nadir + 1.2 * lmaj_ * orientation_, i);
  }
  setLineWidth(minLineWidth_);
  for (i = 0; i != scale_->minors_p.size(); ++i) 
  {
    double t = (scale_->minors_p[i] - start_) / (stop_-start_);
    nadir = beg_ + t * runningpoint;
    minorpos_.push_back(drawTic(nadir, lmin_));
  }
}

void Axis::drawTicLabel(Triple pos, int mtic)
{
  if (!drawNumbers_ || (mtic < 0))
      return;
    
  markerLabel_[mtic].setFont(numberfont_.family(), numberfont_.pointSize(), numberfont_.weight(), numberfont_.italic());
  markerLabel_[mtic].setColor(numbercolor_);
  markerLabel_[mtic].setString(scale_->ticLabel(mtic));   
  markerLabel_[mtic].setPosition(pos, scaleNumberAnchor_);
  markerLabel_[mtic].adjust(numbergap_);
  markerLabel_[mtic].draw();
}

Triple Axis::drawTic(Triple nadir, double length)
{
  double ilength = (symtics_) ? -length : 0.0;

  glBegin( GL_LINES );
  glColor4d(color.r,color.g,color.b,color.a);
  glVertex3d( nadir.x  + ilength * orientation_.x,
              nadir.y  + ilength * orientation_.y,
              nadir.z  + ilength * orientation_.z) ; 
  glVertex3d( nadir.x  + length * orientation_.x,
              nadir.y  + length * orientation_.y,
              nadir.z  + length * orientation_.z);
  glEnd();
  return nadir;
}

void Axis::setNumberFont(QString const& family, int pointSize, int weight, bool italic)
{
  numberfont_ = QFont(family, pointSize, weight, italic );
}

void Axis::setNumberFont(QFont const& font)
{
  numberfont_ = font;
}

void Axis::setNumberColor(RGBA col)
{
  numbercolor_ = col;
}

void Axis::setLabelFont(QString const& family, int pointSize, int weight, bool italic)
{
  labelfont_ = QFont(family, pointSize, weight, italic );
  label_.setFont(family, pointSize, weight, italic);
}

void Axis::setLabelFont(QFont const& font)
{
  setLabelFont(font.family(), font.pointSize(), font.weight(), font.italic());
}

void Axis::setLabelString(QString const& name)
{
  label_.setString(name);
}

/*!
  Sets label position in conjunction with an anchoring strategy
*/
void Axis::setLabelPosition(const Triple& pos,Qwt3D::ANCHOR an)
{
  label_.setPosition(pos, an);
}

//! Sets color for label
void Axis::setLabelColor(RGBA col)
{
  label_.setColor(col);
}

/*! 
  This variant sets a user-defined scale object.
  Use with a heap based initialized pointer only.
  The axis adopts ownership. 
*/
void Axis::setScale(Qwt3D::Scale* val)
{
  scale_ = ValuePtr<Scale>(val); 
}

/*!
  Sets one of the predefined scaling types.
  \warning Too small intervals in logarithmic scales lead to  
  empty scales (or perhaps a scale only containing an isolated 
  major tic). Better switch to linear scales in such cases.
*/
void Axis::setScale(Qwt3D::SCALETYPE val)
{
  switch(val) {
  case Qwt3D::LINEARSCALE:
    setScale(new LinearScale);
    break;
  case Qwt3D::LOG10SCALE:
    setScale(new LogScale);
    setMinors(9);
    break;
  default:
    break;
  }
}
